op 	 {
  name: "BoostedTreesCalculateBestFeatureSplit"
  input_arg {
    name: "node_id_range"
    type: DT_INT32
  }
  input_arg {
    name: "stats_summary"
    type: DT_FLOAT
  }
  input_arg {
    name: "l1"
    type: DT_FLOAT
  }
  input_arg {
    name: "l2"
    type: DT_FLOAT
  }
  input_arg {
    name: "tree_complexity"
    type: DT_FLOAT
  }
  input_arg {
    name: "min_node_weight"
    type: DT_FLOAT
  }
  output_arg {
    name: "node_ids"
    type: DT_INT32
  }
  output_arg {
    name: "gains"
    type: DT_FLOAT
  }
  output_arg {
    name: "feature_dimensions"
    type: DT_INT32
  }
  output_arg {
    name: "thresholds"
    type: DT_INT32
  }
  output_arg {
    name: "left_node_contribs"
    type: DT_FLOAT
  }
  output_arg {
    name: "right_node_contribs"
    type: DT_FLOAT
  }
  output_arg {
    name: "split_with_default_directions"
    type: DT_STRING
  }
  attr {
    name: "logits_dimension"
    type: "int"
    has_minimum: true
    minimum: 1
  }
  attr {
    name: "split_type"
    type: "string"
    default_value {
      s: "inequality"
    }
    allowed_values {
      list {
        s: "inequality"
        s: "equality"
      }
    }
  }
}
